#!/bin/bash

# 设置HugeGraph服务器地址和图名称
HUGEGRAPH_URL="http://localhost:8080"
GRAPH_NAME="hugegraph"

# 清理旧的错误日志（如果存在）
rm -rf loader/vertex_import/failure-data 2>/dev/null
rm -rf loader/edge_import/failure-data 2>/dev/null

# 确保schema已经正确创建
echo "=== 确保schema已正确创建 ==="
java -cp src/main/java com.meiya.hugegraph.schema.TransactionSchema

# 步骤1：导入顶点数据
echo "=== 开始导入顶点数据 ==="
hugegraph-loader -g $GRAPH_NAME -f loader/vertex_import.json -h $HUGEGRAPH_URL

# 检查顶点数量
echo "顶点数据导入完成，查询顶点数量..."
hugegraph-client -g $GRAPH_NAME -e "g.V().count()"

echo "等待5秒后开始导入边数据..."
sleep 5

# 步骤2：导入边数据 - 使用简化配置并禁用顶点检查
echo "=== 开始导入边数据 ==="
# 注意：--check-vertex false 参数允许创建指向不存在顶点的边
hugegraph-loader -g $GRAPH_NAME -f loader/edge_import.json -h $HUGEGRAPH_URL --check-vertex false

echo "=== 数据导入完成 ==="

# 输出数据统计信息
echo "=== 数据统计 ==="
echo "顶点数量："
hugegraph-client -g $GRAPH_NAME -e "g.V().count()"
echo "边数量："
hugegraph-client -g $GRAPH_NAME -e "g.E().count()"
echo "张三卡数量："
hugegraph-client -g $GRAPH_NAME -e "g.V().has('holderName', textContains('张三')).count()" 